package top.abilities.util.util;

import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Component;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
import redis.clients.jedis.params.SetParams;

import javax.annotation.PostConstruct;
import java.util.concurrent.locks.ReentrantLock;

@Component
public class RedisUtil {

    @Autowired
    Environment env;

    private static Environment envStatic;

    @PostConstruct
    public void initEnv() {
        envStatic = env;
    }

    protected static ReentrantLock lockPool = new ReentrantLock();
    protected static ReentrantLock lockJedis = new ReentrantLock();

    private static Logger _log = LoggerFactory.getLogger(RedisUtil.class);

//    // Redis服务器IP
//    private static String IP = envStatic.getProperty("master.redis.ip");
//
//    // Redis的端口号
//    private static int PORT = PropertiesFileUtil.getInstance("application-dev").getInt("master.redis.port");
//
//    // 访问密码
//    private static String PASSWORD = AESUtil.AESDecode(PropertiesFileUtil.getInstance("application-dev").get("master.redis.password"));
//    // 可用连接实例的最大数目，默认值为8；
//    // 如果赋值为-1，则表示不限制；如果pool已经分配了maxActive个jedis实例，则此时pool的状态为exhausted(耗尽)。
//    private static int MAX_ACTIVE = PropertiesFileUtil.getInstance("application-dev").getInt("master.redis.max_active");
//
//    // 控制一个pool最多有多少个状态为idle(空闲的)的jedis实例，默认值也是8。
//    private static int MAX_IDLE = PropertiesFileUtil.getInstance("application-dev").getInt("master.redis.max_idle");
//
//    // 等待可用连接的最大时间，单位毫秒，默认值为-1，表示永不超时。如果超过等待时间，则直接抛出JedisConnectionException；
//    private static int MAX_WAIT = PropertiesFileUtil.getInstance("application-dev").getInt("master.redis.max_wait");
//
//    // 超时时间
//    private static int TIMEOUT = PropertiesFileUtil.getInstance("application-dev").getInt("master.redis.timeout");

    // 在borrow一个jedis实例时，是否提前进行validate操作；如果为true，则得到的jedis实例均是可用的；
    private static boolean TEST_ON_BORROW = false;

    private static JedisPool jedisPool = null;

    /**
     * redis过期时间,以秒为单位
     */
    public final static int EXRP_HOUR = 60 * 60;            //一小时
    public final static int EXRP_DAY = 60 * 60 * 24;        //一天
    public final static int EXRP_MONTH = 60 * 60 * 24 * 30;    //一个月

    /**
     * 初始化Redis连接池
     */
    private static void initialPool() {
        try {
            JedisPoolConfig config = new JedisPoolConfig();
            config.setMaxTotal(Integer.valueOf(envStatic.getProperty("master.redis.max_active")));
            config.setMaxIdle(Integer.valueOf(envStatic.getProperty("master.redis.max_idle")));
            config.setMaxWaitMillis(Integer.valueOf(envStatic.getProperty("master.redis.max_wait")));
            config.setTestOnBorrow(TEST_ON_BORROW);
            jedisPool = new JedisPool(config, envStatic.getProperty("master.redis.ip"), Integer.valueOf(envStatic.getProperty("master.redis.port")), Integer.valueOf(envStatic.getProperty("master.redis.timeout")));
			/*if(jedisPool){

			}*/
        } catch (Exception e) {
            e.printStackTrace();
            _log.error("First create JedisPool error : " + e);
        }
    }

    /**
     * 在多线程环境同步初始化
     */
    private static synchronized void poolInit() {
        if (null == jedisPool) {
            initialPool();
        }
    }

    /**
     * 同步获取Jedis实例
     *
     * @return Jedis
     */
    public synchronized static Jedis getJedis() {
        poolInit();
        Jedis jedis = null;
        try {
            if (null != jedisPool) {
                jedis = jedisPool.getResource();
                try {
                    jedis.auth(AESUtil.AESDecode(envStatic.getProperty("master.redis.password")));
                    jedis.select(Integer.valueOf(envStatic.getProperty("master.redis.database")));
//					jedis.auth(PASSWORD);
                } catch (Exception e) {
                    e.printStackTrace();
                    throw e;
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            _log.error("Get jedis error : " + e);
        }
        return jedis;
    }

    /**
     * 设置 String
     *
     * @param key
     * @param value
     */
    public synchronized static void set(String key, String value) {
        try {
            value = StringUtils.isBlank(value) ? "" : value;
            Jedis jedis = getJedis();
            jedis.set(key, value);
            jedis.close();
        } catch (Exception e) {
            _log.error("Set key error : " + e);
        }
    }

    /**
     * 设置 byte[]
     *
     * @param key
     * @param value
     */
    public synchronized static void set(byte[] key, byte[] value) {
        try {
            Jedis jedis = getJedis();
            jedis.set(key, value);
            jedis.close();
        } catch (Exception e) {
            _log.error("Set key error : " + e);
        }
    }

    /**
     * 设置 String 过期时间
     *
     * @param key
     * @param value
     * @param seconds 以秒为单位
     */
    public synchronized static void set(String key, String value, int seconds) {
        try {
            value = StringUtils.isBlank(value) ? "" : value;
            Jedis jedis = getJedis();
            jedis.setex(key, seconds, value);
            jedis.close();
        } catch (Exception e) {
            _log.error("Set keyex error : " + e);
        }
    }

    public synchronized static boolean tryLock(String key, SetParams setParams) {
        String ret = null;
        try {
            Jedis jedis = getJedis();
            ret = jedis.set(key, "", setParams);
            close(jedis);
        } catch (Exception e) {
            _log.error("tryLock error : " + e);
        }
        if ("OK".equalsIgnoreCase(ret)) {
            return true;
        }
        return false;
    }

    /**
     * 设置 byte[] 过期时间
     *
     * @param key
     * @param value
     * @param seconds 以秒为单位
     */
    public synchronized static void set(byte[] key, byte[] value, int seconds) {
        try {
            Jedis jedis = getJedis();
            jedis.set(key, value);
            jedis.expire(key, seconds);
            jedis.close();
        } catch (Exception e) {
            _log.error("Set key error : " + e);
        }
    }

    /**
     * 获取String值
     *
     * @param key
     * @return value
     */
    public synchronized static String get(String key) {
        Jedis jedis = getJedis();
        if (null == jedis) {
            return null;
        }
        String value = jedis.get(key);
        close(jedis);
        return value;
    }

    /**
     * 获取byte[]值
     *
     * @param key
     * @return value
     */
    public synchronized static byte[] get(byte[] key) {
        Jedis jedis = getJedis();
        if (null == jedis) {
            return null;
        }
        byte[] value = jedis.get(key);
        jedis.close();
        return value;
    }

    /**
     * 删除值
     *
     * @param key
     */
    public synchronized static void remove(String key) {
        try {
            Jedis jedis = getJedis();
            jedis.del(key);
            jedis.close();
        } catch (Exception e) {
            _log.error("Remove keyex error : " + e);
        }
    }

    /**
     * 删除值
     *
     * @param key
     */
    public synchronized static void remove(byte[] key) {
        try {
            Jedis jedis = getJedis();
            jedis.del(key);
            jedis.close();
        } catch (Exception e) {
            _log.error("Remove keyex error : " + e);
        }
    }

    /**
     * 判断key是否存在
     *
     * @param key
     * @return
     */
    public synchronized static boolean exists(byte[] key) {
        Jedis jedis = getJedis();
        if (null == jedis) {
            _log.info("exists info : jedis is null");
            return false;
        }
        boolean value = jedis.exists(key);
        jedis.close();
        return value;
    }

    /**
     * 判断key是否存在
     *
     * @param key
     * @return
     */
    public synchronized static boolean exists(String key) {
        Jedis jedis = getJedis();
        if (null == jedis) {
            _log.info("exists info : jedis is null");
            return false;
        }
        boolean value = jedis.exists(key);
        jedis.close();
        return value;
    }

    /**
     * 存储redis队列，顺序存储
     * lpush
     *
     * @param key
     * @param key
     */
    public synchronized static void lpush(String key, String... strings) {
        try {
            Jedis jedis = RedisUtil.getJedis();
            jedis.lpush(key, strings);
            jedis.close();
        } catch (Exception e) {
            _log.error("lpush error : " + e);
        }
    }

    /**
     * lrem
     *
     * @param key
     * @param count
     * @param value
     */
    public synchronized static void lrem(String key, long count, String value) {
        try {
            Jedis jedis = RedisUtil.getJedis();
            jedis.lrem(key, count, value);
            jedis.close();
        } catch (Exception e) {
            _log.error("lpush error : " + e);
        }
    }

    /**
     * sadd
     *
     * @param key
     * @param value
     * @param seconds
     */
    public synchronized static void sadd(String key, String value, int seconds) {
        try {
            Jedis jedis = RedisUtil.getJedis();
            jedis.sadd(key, value);
            jedis.expire(key, seconds);
            jedis.close();
        } catch (Exception e) {
            _log.error("sadd error : " + e);
        }
    }

    /**
     * incr
     *
     * @param key
     * @return value
     */
    public synchronized static Long incr(String key) {
        Jedis jedis = getJedis();
        if (null == jedis) {
            return null;
        }
        long value = jedis.incr(key);
        jedis.close();
        return value;
    }

    /**
     * decr
     *
     * @param key
     * @return value
     */
    public synchronized static Long decr(String key) {
        Jedis jedis = getJedis();
        if (null == jedis) {
            return null;
        }
        long value = jedis.decr(key);
        jedis.close();
        return value;
    }

    public synchronized static Long setnx(String key, String value) {
        Jedis jedis = getJedis();
        if (null == jedis) {
            return null;
        }
        long i = jedis.setnx(key, value);
        close(jedis);
        return i;
    }


    public synchronized static String getset(String key, String value) {
        Jedis jedis = getJedis();
        if (null == jedis) {
            return null;
        }
        String oldValue = jedis.getSet(key, value);
        close(jedis);
        return oldValue;
    }

    public synchronized static Long del(String key) {
        Jedis jedis = getJedis();
        if (null == jedis) {
            return null;
        }
        Long i = jedis.del(key);
        close(jedis);
        return i;
    }

    private static void close(Jedis jedis) {
        jedis.close();
        if (jedis.isConnected()) {
            try {
                System.out.println("退出" + jedis.toString() + ":" + jedis.quit());
                jedis.disconnect();
            } catch (Exception e) {
                System.out.println("退出失败");
                e.printStackTrace();
            }
        }
    }
}
